import numpy as np
import os


data = np.load(f"/home/fit/liuyebin/WORK/lingweidang/codes/hoi4d/GeometryCrafter/outputs/color_video_416_624_00002.npz")
pred_pointmap = data["point_map"]
pred_mask = data["mask"]
print(f"pred_pointmap: {pred_pointmap.shape}, {pred_pointmap[..., 0].min(), pred_pointmap[..., 0].max()}, {pred_pointmap[..., 1].min(), pred_pointmap[..., 1].max()}, {pred_pointmap[..., 2].min(), pred_pointmap[..., 2].max()}")
print(f"pred_mask: {pred_mask.shape}, {pred_mask.min(), pred_mask.max()}")